package com.shujia.flink.window

import org.apache.flink.streaming.api.TimeCharacteristic
import org.apache.flink.streaming.api.scala._
import org.apache.flink.streaming.api.windowing.assigners.{EventTimeSessionWindows, ProcessingTimeSessionWindows}
import org.apache.flink.streaming.api.windowing.time.Time

object Demo2SessionWindow {
  def main(args: Array[String]): Unit = {
    /**
      * 会话窗口： 一段时间没有数据生成一个窗口
      *
      */


    val env: StreamExecutionEnvironment = StreamExecutionEnvironment.getExecutionEnvironment


    //当有多个并行度的时候需要所有并行度中的水位线对齐之后才能计算
    env.setParallelism(1)


    //设置时间模式
    env.setStreamTimeCharacteristic(TimeCharacteristic.EventTime)

    val linesDS: DataStream[String] = env.socketTextStream("master", 8888)

    val eventDS: DataStream[(String, Long)] = linesDS.map(line => {
      val split: Array[String] = line.split(",")
      (split(0), split(1).toLong)
    })

    //指定时间字段
    val assDS: DataStream[(String, Long)] = eventDS.assignAscendingTimestamps(_._2)

    /**
      * ProcessingTimeSessionWindows: 处理时间的会话窗口
      * EventTimeSessionWindows： 事件时间的会话窗口，需要设置时间模式和时间字段，水位线
      *
      */

    assDS
      .map(kv => (kv._1, 1))
      .keyBy(_._1)
      //10秒没有数据开始计算, 针对每一个key来说的
      .window(EventTimeSessionWindows.withGap(Time.seconds(10)))
      .sum(1)
      .print()

    env.execute()

  }

}
